# 重新保存模型以消除版本不一的警告
# 解决方案见https://github.com/tbepler/protein-sequence-embedding-iclr2019/issues/17
if __name__ == '__main__':
    import sys
    if len(sys.argv) > 1:
        import torch
        path = sys.argv[1]
        model = torch.load(path, map_location=torch.device('cpu'))
        torch.save(model, path)